import torch

device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')  # sets device for model and PyTorch tensors

letters = 'abcdefghijklmnopqrstuvwxyz'

letter_to_id = {s: i for i, s in enumerate(letters)}
id_to_letter = {i: s for i, s in enumerate(letters)}

batch_size = 32
